import time
import torch

# device
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

bc_model_path = "/home/model_zoo/LLM/baichuan-inc/Baichuan2-13B-Chat"
url = ""  # 生产环境
user_id = ""  # 该成自己的id, 如
user_id_list = ["shilin.zhuang", "", "", ""]
server_path = ""
server_id = ""  # 找 索取
# 每个application或者service都会有一对app_id和app_secret
endpoint = url  # "YOUR_APP_ENDPOINT"
app_secret = ''  # "YOUR_APP_SECRET"  # 从 ... 申请的
app_id = ''  # 'YOUR_APP_ID'  #

headers = {"content-type": "application/json",
           "X-USER-ID": user_id,  # 使用自己的 user_id:
           "X-SERVER-ID": server_id
           }

# NIO-GPT model
path = server_path  # "API_PATH"
payload = {
    'app_id': app_id,
    'timestamp': int(time.time()),
    'sign': ''
}
tmp = payload.copy()
target_url = url + server_path

# VALL-E-X model
checkpoints_dir = "/home/shilin.zhuang.o/VALL-E-X/checkpoints/"
model_checkpoint_name = "vallex-checkpoint.pt"
wav_save_path = '/home/model_zoo/VALL-E-X/generated_files/'

# logo_path
bc_logo_path = "/workspace/chatall/logo/baichuan-logo.png"
niogpt_logo_path = "/workspace/chatall/logo/chatgpt-35-logo.png"
chatglm_logo_path = "/workspace/chatall/logo/chatglm-logo.png"
